#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import argparse

from utils.dataset_utils import init_dataset


def parse_opt():
    parser = argparse.ArgumentParser()
    
    # ------------------------------------------------------------- #
    # PTL dataset options
    # ------------------------------------------------------------- #
    parser.add_argument('--exp_id', type=str, default='exp1')
    parser.add_argument('--syn_dataset_root', type=str, default='../datasets/synthetic/new/')
    parser.add_argument('--syn_chars', nargs='+', type=str, default=['juliet', 'kelly', 'lucy', 'mary', 'romeo', 'scott', 'troy', 'victor'])
    parser.add_argument('--syn_poses', nargs='+', type=str, default=['', 'prone', 'squat'])
    parser.add_argument('--syn_smp_angles', type=int, default=30)
    parser.add_argument('--syn_min_wh', type=int, default=0)
    parser.add_argument('--real_dataset_root', type=str, default='../datasets/visdrone')
    parser.add_argument('--real_dataset_format', type=str, choices=['vis', 'oku', 'icg'], default='vis')
    parser.add_argument('--real_dataset_spl_type', type=str, choices=['all', 'per', 'abs'], default='abs')
    parser.add_argument('--real_dataset_spl_num', type=int, default=20)
    
    return parser.parse_args()   


def main(opt):
    # Dataset initialization
    init_dataset(opt)


if __name__ == "__main__":
    opt = parse_opt()
    main(opt)